import databricks.koalas as ks
import numpy as np
import pandas as pd
import pyspark.pandas as ps

if __name__ == '__main__':
    dates = pd.date_range("20130101", periods=6)
    print(type(dates), dates)
    pdf = pd.DataFrame(np.random.randn(6, 4), index=dates, columns=list('ABCD'))
    print(type(pdf), pdf)
    pdf.index = pdf.index.astype('str')
    # kdf = ks.from_pandas(pdf)
    kdf = ps.from_pandas(pdf)
    print(type(kdf), kdf)
